-
Notifications
You must be signed in to change notification settings - Fork 480
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Don't fallback for pow #5919
Don't fallback for pow #5919
Conversation
752511c
to
e00bd7f
Compare
op info test started to complain, can you take a look? |
e00bd7f
to
2af9b6e
Compare
2af9b6e
to
a09ecee
Compare
@@ -418,4 +419,5 @@ def test_reference_eager(self, device, dtype, op): | |||
instantiate_device_type_tests(TestOpInfo, globals()) | |||
|
|||
if __name__ == '__main__': | |||
run_tests() | |||
#run_tests() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nit, remove this
test/test_core_aten_ops.py
Outdated
@@ -3295,6 +3295,29 @@ def test_aten_pow_Tensor_Scalar_2(self): | |||
kwargs = dict() | |||
run_export_and_compare(self, torch.ops.aten.pow.Tensor_Scalar, args, kwargs) | |||
|
|||
def test_aten_pow_Scalar_1(self): | |||
args = (10000, | |||
torch.Tensor([ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
any reason not using randn here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
either way works.
* Don't fallback for pow
* Don't fallback for pow
* Don't fallback for pow
* Don't fallback for pow
No description provided.